from django.urls import path
from torch.nn.functional import dropout

from .views import MnistPredictView, MnistPredictDropoutView

urlpatterns = [
    path('predict/', MnistPredictView.as_view(), name='mnist_predict'),

    path('predict_dropout/', MnistPredictDropoutView.as_view(), name='mnist_predict_dropout'),
]